package com.yb.project.algorithm.practice.od2;

/**
 * 2079. 给植物浇水
 *
 * 你打算用一个水罐给花园里的 n 株植物浇水。植物排成一行，从左到右进行标记，编号从 0 到 n - 1 。
 * 其中，第 i 株植物的位置是 x = i 。x = -1
 * 处有一条河，你可以在那里重新灌满你的水罐。
 * 每一株植物都需要浇特定量的水。你将会按下面描述的方式完成浇水：
 * 按从左到右的顺序给植物浇水。 在给当前植物浇完水之后，如果你没有足够的水 完全 浇灌下一株植物，那么你就需要返回河边重新装满水罐。
 * 你 不能 提前重新灌满水罐。 最初，你在河边（也就是，x = -1），在 x 轴上每移动 一个单位 都需要 一步 。
 * 给你一个下标从 0 开始的整数数组 plants ，数组由 n 个整数组成。其中，plants[i] 为第 i 株植物需要的水量。另有一个整数 capacity
 * 表示水罐的容量，返回浇灌所有植物需要的 步数 。
 */
public class Leetcode2079 {
    public static void main(String[] args) {
        //
        Leetcode2079 yb = new Leetcode2079();
        int[] plants = new int[]{3,2,4,2,1};
        System.out.println(yb.wateringPlants(plants,6));
    }

    public int wateringPlants(int[] plants, int capacity) {
        int leftover = capacity;
        int res = 0;
        for(int i=0;i<plants.length;i++){
            res+=1;
            leftover = leftover - plants[i];
            if(i<plants.length-1 && leftover<plants[i+1]){
                res=res+(i+1)*2;
                leftover = capacity;
            }
        }
        return res;
    }
}
